from selenium.webdriver.common.by import By
from selenium import webdriver
from Yuekao.Page.Page import Page
import unittest,time,logging

class TestCase(unittest.TestCase):

    def setUp(self):
        logging.info('start')
        # 打开指定浏览器
        self.driver = webdriver.Edge()
        # 窗口最大化
        self.driver.maximize_window()

    def tearDown(self):
        logging.info('end')
        # 退出浏览器
        self.driver.quit()

    # 百度搜索功能
    def test_01_serach(self):
        # 实例化Page类
        po = Page(self.driver)
        # 打开百度网址
        po.open_url('https://www.baidu.com/')
        # 搜索框输入
        po.send_search()
        # 点击搜索按钮
        po.click_search()
        # 隐式等待断言
        self.driver.implicitly_wait(3)
        # 断言百度首页文本
        dy = self.driver.find_element(By.CLASS_NAME,'toindex').text
        self.assertEqual('百度首页',dy)

    def test_02_news(self):
        # 实例化Page类
        po = Page(self.driver)
        # 打开百度网址
        po.open_url('https://www.baidu.com/')
        # 点击新闻文本
        po.click_news()
        # 点击热搜
        po.click_hot()
        # 进入详情
        po.click_more()
        # 强制等待断言
        time.sleep(3)
        # 断言央视新闻
        dy = self.driver.find_element(By.CLASS_NAME,'_2gGWi').text
        self.assertEqual('央视新闻',dy)

    def test_03_input(self):
        # 实例化Page类
        po = Page(self.driver)
        # 打开百度网址
        po.open_url('https://www.baidu.com/')
        # 搜索框输入
        po.send_search()
        # 点击搜索按钮
        po.click_search()
        # 隐式等待断言
        self.driver.implicitly_wait(3)
        # 断言百度首页文本
        dy = self.driver.find_element(By.CLASS_NAME,'toindex').text
        self.assertEqual('百度首页',dy)